from flask import jsonify
from datetime import datetime

def parse_date(date_str):
    """解析日期字符串为datetime对象"""
    if not date_str:
        return None
    
    try:
        # 尝试解析ISO格式
        return datetime.fromisoformat(date_str.replace('Z', '+00:00'))
    except ValueError:
        try:
            # 尝试解析YYYY-MM-DD格式
            return datetime.strptime(date_str, '%Y-%m-%d')
        except ValueError:
            try:
                # 尝试解析YYYY-MM-DD HH:MM:SS格式
                return datetime.strptime(date_str, '%Y-%m-%d %H:%M:%S')
            except ValueError:
                return None

def format_response(code, msg, data):
    """格式化API响应"""
    return jsonify({
        "code": code,
        "msg": msg,
        "data": data
    })